﻿// This source code is dual-licensed under the Apache License, version
// 2.0, and the Mozilla Public License, version 2.0.
//
// The APL v2.0:
//
//---------------------------------------------------------------------------
//   Copyright (c) 2007-2025 Broadcom. All Rights Reserved.
//
//   Licensed under the Apache License, Version 2.0 (the "License");
//   you may not use this file except in compliance with the License.
//   You may obtain a copy of the License at
//
//       https://www.apache.org/licenses/LICENSE-2.0
//
//   Unless required by applicable law or agreed to in writing, software
//   distributed under the License is distributed on an "AS IS" BASIS,
//   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//   See the License for the specific language governing permissions and
//   limitations under the License.
//---------------------------------------------------------------------------
//
// The MPL v2.0:
//
//---------------------------------------------------------------------------
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at https://mozilla.org/MPL/2.0/.
//
//  Copyright (c) 2007-2025 Broadcom. All Rights Reserved.
//---------------------------------------------------------------------------

using System;
using System.IO;
using System.Text;
using Xunit.Abstractions;

namespace Test
{
    public class TestOutputWriter : TextWriter
    {
        private readonly ITestOutputHelper _output;
        private readonly string _testDisplayName;

        public TestOutputWriter(ITestOutputHelper output, string testDisplayName)
        {
            _output = output;
            _testDisplayName = testDisplayName;
        }

        public override Encoding Encoding => Encoding.UTF8;

        public override void Write(char[] buffer, int index, int count)
        {
            if (count > 2)
            {
                var sb = new StringBuilder(Util.Now);
                sb.Append(" [DEBUG] ");
                sb.Append(_testDisplayName);
                sb.Append(" | ");
                sb.Append(buffer, index, count);
                try
                {
                    _output.WriteLine(sb.ToString().TrimEnd());
                }
                catch (InvalidOperationException)
                {
                    /*
                     * Note:
                     * This exception can be thrown if there is no running test.
                     * Catch it here to prevent it causing an incorrect test failure.
                     */
                }
            }
        }
    }
}
